from setuptools import setup
from pybind11.setup_helpers import Pybind11Extension, build_ext

ext_modules = [
    Pybind11Extension(
        "cpu_extension",
        ["main.cpp", "hello.cpp"],
    ),
]

setup(
    name="cpu_hello",
    ext_modules=ext_modules,
    cmdclass={"build_ext": build_ext},
)